package com.example.offer;

import com.example.structure.TreeNode;

/**
 * 剑指 Offer 55 - II. 平衡二叉树
 * 输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。
 * <p>
 * 示例 1:
 * <p>
 * 给定二叉树 [3,9,20,null,null,15,7]
 * <p>
 * 3
 * / \
 * 9  20
 * /  \
 * 15   7
 * 返回 true 。
 * <p>
 * 示例 2:
 * <p>
 * 给定二叉树 [1,2,2,3,3,null,null,4,4]
 * <p>
 * 1
 * / \
 * 2   2
 * / \
 * 3   3
 * / \
 * 4   4
 * 返回 false 。
 */
public class BalanceTree {
    public static int lengthOfLastWord(String s) {
        String trim = s.trim();
        int n = trim.length();
        int start = 0;
        for (int i = n - 1; i >= 0; i--) {
            if (trim.charAt(i) == ' ') {
                start = i;
                break;
            }
        }
        return n - 1 - start;
    }

    public static void main(String[] args) {
        lengthOfLastWord("   fly me   to   the moon  ");
    }

    public boolean isBalanced(TreeNode root) {
        return height(root) >= 0;
    }

    public int height(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int leftHeight = height(root.left);
        int rightHeight = height(root.right);
        //如果出现一棵不平衡的，那么一定不平衡
        if (leftHeight == -1 || rightHeight == -1 || Math.abs(leftHeight - rightHeight) > 1) {
            return -1;
        } else {
            //否则返回当前树的最大高度
            return Math.max(leftHeight, rightHeight) + 1;
        }
    }
}

